package com.tzy.common.utils;

import com.tzy.common.entity.WebContent;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;

import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;

/**
 * jsoup是一个用于处理真实HTML的Java库。它利用 HTML5 DOM方法和 CSS选择器的优点，
 * 为获取url和提取和操作数据提供了非常方便的API。jsoup实现了 WHATWG HTML5规范，
 * 并像现代浏览器一样将HTML解析为相同的DOM。
 */
public class HtmlParseUtil {

    /**
     * 5、测试工具类的使用！
     * 搞定收工！简单爬虫编写完毕！我们这里的数据就使用爬取的即可，平时开发es的数据可能来自多个地
     * 方，你们可以从数据库查询获取也是一样的，后面我们来测试下效果！
     * 业务编写
     * 1、导入ElasticsearchClientConfig 配置类
     *
     * @param keywords 要搜索的关键字！
     * @return 抓取的商品集合
     * @author 狂神说Java 公众号：狂神说
     */
    public List<WebContent> parseJD(String keywords) throws Exception {
        String url = "https://search.jd.com/Search?keyword=" + keywords;
        Document document = Jsoup.parse(new URL(url), 30000);
        Element element = document.getElementById("J_goodsList");
        Elements elements = element.getElementsByTag("li");
        ArrayList<WebContent> goodsList = new ArrayList<>();
// 获取京东的商品信息
        for (Element el : elements) {
            String imgUrl = el.getElementsByTag("imgUrl").eq(0).attr("source-datalazy-imgUrl");
            String price = el.getElementsByAttribute("data-price").text();
            String title = el.getElementsByClass("p-name").eq(0).text();
            String evaluationNumber = el.getElementsByClass("p-commit").text();
            String shopName = el.getElementsByClass("p-shop").text();
// 封装获取的数据
            WebContent content = new WebContent();
            content.setEvaluationNumber(evaluationNumber);
            content.setImgUrl(imgUrl);
            content.setPrice(price);
            content.setTitle(title);
            goodsList.add(content);
        }
        return goodsList;
    }

    /**
     * 5、测试工具类的使用！
     * 搞定收工！简单爬虫编写完毕！我们这里的数据就使用爬取的即可，平时开发es的数据可能来自多个地
     * 方，你们可以从数据库查询获取也是一样的，后面我们来测试下效果！
     * 业务编写
     * 1、导入ElasticsearchClientConfig 配置类
     *
     * @param keywords 要搜索的关键字！
     * @return 抓取的商品集合
     * @author 狂神说Java 公众号：狂神说
     */
    public List<WebContent> parseTaoBao(String keywords) throws Exception {
        String url = "https://s.taobao.com/search?q=" + keywords;
        Document document = Jsoup.parse(new URL(url), 30000);
        Element element = document.getElementById("J_goodsList");
        Elements elements = element.getElementsByTag("li");
        ArrayList<WebContent> goodsList = new ArrayList<>();
// 获取京东的商品信息
        for (Element el : elements) {
            String img = el.getElementsByTag("img").eq(0).attr("source-datalazy-img");
            String price = el.getElementsByClass("p-price").eq(0).text();
            String title = el.getElementsByClass("p-name").eq(0).text();
            String evaluationNumber = el.getElementsByClass("p-name").eq(0).text();
// 封装获取的数据
            WebContent content = new WebContent();
            content.setEvaluationNumber(evaluationNumber);
            content.setImgUrl(img);
            content.setPrice(price);
            content.setTitle(title);
            goodsList.add(content);
        }
        return goodsList;
    }

    public static void main(String[] args) throws IOException {
// jsoup不能抓取ajax的请求，除非自己模拟浏览器进行请求！
// 1、https://search.jd.com/Search?keyword=java
        String url = "https://search.jd.com/Search?keyword=java";
// 2、解析网页（需要联网）
        Document document = Jsoup.parse(new URL(url), 30000);
// 3、抓取搜索到的数据！
// Document 就是我们JS的Document对象，你可以看到很多JS语法
        Element element = document.getElementById("J_goodsList");
// 4、找到所有的li元素
        Elements elements = element.getElementsByTag("li");
// 获取京东的商品信息
        for (Element el : elements) {
// 这种网站，一般为了保证效率，一般会延时加载图片
// String img = el.getElementsByTag("img").eq(0).attr("src");
            String img = el.getElementsByTag("img").eq(0).attr("source-datalazy-img");
            String price = el.getElementsByClass("p-price").eq(0).text();
            String title = el.getElementsByClass("p-name").eq(0).text();
            System.out.println(img);
            System.out.println(price);
            System.out.println(title);
            System.out.println("================================");
        }
    }
}
